# main.py
from gradio_ui import GradioUI
from llm_integration import LLMIntegration
from agents import Role

if __name__ == "__main__":
    # 初始化 LLM 集成
    # llm = LLMIntegration(api_key="YOUR_OPENAI_API_KEY")
    #
    # # 在 Role 类的 interact_with_llm 方法中使用 llm.generate_response
    # def patch_interact_with_llm():
    #     def new_interact_with_llm(self, prompt):
    #         return llm.generate_response(prompt)
    #     Role.interact_with_llm = new_interact_with_llm
    #
    # patch_interact_with_llm()

    # 启动 Gradio 界面
    ui = GradioUI()
    ui.create_ui()